package database

import (
	"fmt"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"log"
	"one-wiki/config"
)

var db *gorm.DB
var DB *gorm.DB

// Init 初始化数据库连接
func Init() {
	var err error
	dsn := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
		config.DatabaseConfig.User,
		config.DatabaseConfig.Password,
		config.DatabaseConfig.Host,
		config.DatabaseConfig.Name)

	db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		log.Panicf(err.Error())
	}

	// 配置最大连接数量和最大空闲连接数量
	sqlDB, err := db.DB()
	if err != nil {
		log.Panicf(err.Error())
	}
	sqlDB.SetMaxIdleConns(config.DatabaseConfig.MaxIdleConns)
	sqlDB.SetMaxOpenConns(config.DatabaseConfig.MaxOpenConns)
}
